"""
vector_db_config.py 模块
向量数据库的配置类
"""
from dataclasses import dataclass
import os
from dotenv import load_dotenv

load_dotenv()

@dataclass
class VectorDBConfig:
    """向量数据库配置类"""
    db_type: str = os.getenv("VECTOR_DB_TYPE", "chroma")
    db_path: str = os.getenv("VECTOR_DB_PATH", "./vector_stores/chroma")
    collection_name: str = os.getenv("COLLECTION_NAME", "zbs_agent_kb")
